package com.gitee.ting723.upload;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.time.DateFormatUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.util.Arrays;
import java.util.Calendar;
import java.util.UUID;

/**
 * 文件工具类
 *
 * @author zhanglianwei
 */
@Slf4j
public class FileUtil {


    /**
     * 将图片写入服务器本地文件夹
     *
     * @param file     file
     * @param filename 全路径
     */
    public static void writeFileToLocal(MultipartFile file, String filename) {
        if (!file.isEmpty()) {
            try {
                File filepath = new File(filename);
                if (!filepath.exists()) {
                    boolean isSuccess = filepath.mkdirs();
                    if (!isSuccess) {
                        throw new Exception();
                    }
                }
                // 转存文件
                file.transferTo(new File(filename));
            } catch (Exception e) {
                log.error("upload error", e);
            }
        }
    }

    /**
     * 生成文件全路径
     *
     * @param baseDir 基本目录
     * @param fileExt fileExt
     * @return String
     */
    public static String createFileName(String baseDir, String fileExt) {
        String dateDir = DateFormatUtils.format(Calendar.getInstance(), Constants.DATE_JOINT_FORMAT);
        int rnd = 1000 + (int) (Math.random() * 8999);
        String name = UUID.randomUUID().toString().trim().replaceAll("-", "").substring(0, 6) + rnd;
        return baseDir + dateDir + File.separator + name + fileExt;
    }

    /**
     * 替换windows 落山
     * @param fileName
     * @param baseDir
     * @return
     */
    public static String getUrlName(String fileName, String baseDir) {
        return fileName.replace(baseDir, "").replace(File.separator, Constants.URL_PATH_SEPARATOR);
    }

    /**
     * 生成文件名
     *
     * @param baseDir 基本路径
     * @param file    文件
     * @return 新文件名
     * @throws Exception ex
     */
    public static String createFileName(String baseDir, MultipartFile file) throws Exception {
        String[] fileAllowType = {".gif", ".png", ".jpg", ".jpeg", ".bmp", ".tiff"};
        String fileExt = getFileExt(file);
        if (!Arrays.asList(fileAllowType).contains(fileExt.toLowerCase())) {
            throw new Exception();
        }
        return createFileName(baseDir, fileExt);
    }

    /**
     * 获取扩展名
     *
     * @param file 文件
     * @return file extend name
     * @throws Exception ex
     */
    public static String getFileExt(MultipartFile file) throws Exception {
        String fileOldName = file.getOriginalFilename();
        if (StringUtils.isEmpty(fileOldName)) {
            throw new Exception();
        }
        return fileOldName.substring(fileOldName.lastIndexOf("."));
    }
}

